#include "spi_hard_oled.h"

void spi_hard_config(void)
{
    /* SPI 硬件配置 */
    rcu_periph_clock_enable(RCU_SPI0);
    // 创建SPI结构体
    spi_parameter_struct sps;

    sps.trans_mode = SPI_TRANSMODE_FULLDUPLEX;         // 全双工通讯
    sps.device_mode = SPI_MASTER;                      // 作为主机
    sps.frame_size = SPI_FRAMESIZE_8BIT;               // 数据位宽
    sps.clock_polarity_phase = SPI_CK_PL_LOW_PH_1EDGE; // ckcl = 0   ckph = 0
    sps.nss = SPI_NSS_SOFT;                            // 软件片选（自己拉高拉低）
    // SPI0 在APB2 84MHz 分频32 = 2.625MHz
    sps.prescale = SPI_PSC_32;   // 分频（决定了SPI的时钟速度）
    sps.endian = SPI_ENDIAN_MSB; // 高位优先
    spi_init(SPI0, &sps);

    spi_enable(SPI0);
}
void spi_hard_oled_config(void)
{
    // PA 使能
    rcu_periph_clock_enable(RCU_GPIOA);
    // PA5 PA7 服用输出,上拉电阻
    gpio_mode_set(GPIOA, GPIO_MODE_AF, GPIO_PUPD_PULLUP, GPIO_PIN_5 | GPIO_PIN_7);
    gpio_af_set(GPIOA, GPIO_AF_5, GPIO_PIN_5 | GPIO_PIN_7);

    // PA2,PA3 推挽输出,低电平
    gpio_mode_set(GPIOA, GPIO_MODE_OUTPUT, GPIO_PUPD_NONE, GPIO_PIN_2 | GPIO_PIN_3);
    gpio_output_options_set(GPIOA, GPIO_OTYPE_PP, GPIO_OSPEED_50MHZ, GPIO_PIN_2 | GPIO_PIN_3);
    gpio_bit_reset(GPIOA, GPIO_PIN_2 | GPIO_PIN_3);

    //
    spi_hard_config();
}

u8 OLED_GRAM_SPI_HARD[144][8];

// 反显函数
void OLED_ColorTurn_SPI_HARD(u8 i)
{
    if (i == 0)
    {
        OLED_WR_Byte_SPI_HARD(0xA6, OLED_CMD); // 正常显示
    }
    if (i == 1)
    {
        OLED_WR_Byte_SPI_HARD(0xA7, OLED_CMD); // 反色显示
    }
}

// 屏幕旋转180度
void OLED_DisplayTurn_SPI_HARD(u8 i)
{
    if (i == 0)
    {
        OLED_WR_Byte_SPI_HARD(0xC8, OLED_CMD); // 正常显示
        OLED_WR_Byte_SPI_HARD(0xA1, OLED_CMD);
    }
    if (i == 1)
    {
        OLED_WR_Byte_SPI_HARD(0xC0, OLED_CMD); // 反转显示
        OLED_WR_Byte_SPI_HARD(0xA0, OLED_CMD);
    }
}

void OLED_WR_Byte_SPI_HARD(u8 dat, u8 cmd)
{
    //	是命令还是数据
    if (cmd)
        OLED_DC_Set();
    else
        OLED_DC_Clr();
    // 软件选择设备
    OLED_CS_Clr(); // clr clear  清零  拉低  片选
    //	for(i=0;i<8;i++)
    //	{   // ckpl = 1   ckph=1  第二个沿采样
    //		OLED_SCL_Clr(); // 时钟拉低  更新数据
    //		if(dat&0x80)    // 10000000  MSB
    //		   OLED_SDA_Set();  // 准备数据 1
    //		else
    //		   OLED_SDA_Clr();  // 准备数据 0
    //		OLED_SCL_Set();     // 时钟拉高  发数据
    //		dat<<=1;
    //	}

    // TBE transmit buffer empty  硬件置1，为1表示发送缓冲区为空（可以写，即可以发）
    while (RESET == spi_i2s_flag_get(SPI0, SPI_FLAG_TBE));
    spi_i2s_data_transmit(SPI0, dat);
    while (SET == spi_i2s_flag_get(SPI0, SPI_FLAG_TRANS)); // 等到TRANS标记为0（表示没有传输了）

    OLED_CS_Set(); // 拉高 取消片选
    OLED_DC_Set();
    
}

// 开启OLED显示
void OLED_DisPlay_On_SPI_HARD(void)
{
    OLED_WR_Byte_SPI_HARD(0x8D, OLED_CMD); // 电荷泵使能
    OLED_WR_Byte_SPI_HARD(0x14, OLED_CMD); // 开启电荷泵
    OLED_WR_Byte_SPI_HARD(0xAF, OLED_CMD); // 点亮屏幕
}

// 关闭OLED显示
void OLED_DisPlay_Off_SPI_HARD(void)
{
    OLED_WR_Byte_SPI_HARD(0x8D, OLED_CMD); // 电荷泵使能
    OLED_WR_Byte_SPI_HARD(0x10, OLED_CMD); // 关闭电荷泵
    OLED_WR_Byte_SPI_HARD(0xAE, OLED_CMD); // 关闭屏幕
}

// 更新显存到OLED
void OLED_Refresh_SPI_HARD(void)
{
    u8 i, n;
    for (i = 0; i < 8; i++)
    {
        OLED_WR_Byte_SPI_HARD(0xb0 + i, OLED_CMD); // 设置行起始地址
        OLED_WR_Byte_SPI_HARD(0x00, OLED_CMD);     // 设置低列起始地址
        OLED_WR_Byte_SPI_HARD(0x10, OLED_CMD);     // 设置高列起始地址
        for (n = 0; n < 128; n++)
            OLED_WR_Byte_SPI_HARD(OLED_GRAM_SPI_HARD[n][i], OLED_DATA);
    }
}
// 清屏函数
void OLED_Clear_SPI_HARD(void)
{
    u8 i, n;
    for (i = 0; i < 8; i++)
    {
        for (n = 0; n < 128; n++)
        {
            OLED_GRAM_SPI_HARD[n][i] = 0; // 清除所有数据
        }
    }
    OLED_Refresh_SPI_HARD(); // 更新显示
}

// 画点
// x:0~127
// y:0~63
// t:1 填充 0,清空
void OLED_DrawPoint_SPI_HARD(u8 x, u8 y, u8 t)
{
    u8 i, m, n;
    i = y / 8;
    m = y % 8;
    n = 1 << m;
    if (t)
    {
        OLED_GRAM_SPI_HARD[x][i] |= n;
    }
    else
    {
        OLED_GRAM_SPI_HARD[x][i] = ~OLED_GRAM_SPI_HARD[x][i];
        OLED_GRAM_SPI_HARD[x][i] |= n;
        OLED_GRAM_SPI_HARD[x][i] = ~OLED_GRAM_SPI_HARD[x][i];
    }
}

// 画线
// x1,y1:起点坐标
// x2,y2:结束坐标
void OLED_DrawLine_SPI_HARD(u8 x1, u8 y1, u8 x2, u8 y2, u8 mode)
{
    u16 t;
    int xerr = 0, yerr = 0, delta_x, delta_y, distance;
    int incx, incy, uRow, uCol;
    delta_x = x2 - x1; // 计算坐标增量
    delta_y = y2 - y1;
    uRow = x1; // 画线起点坐标
    uCol = y1;
    if (delta_x > 0)
        incx = 1; // 设置单步方向
    else if (delta_x == 0)
        incx = 0; // 垂直线
    else
    {
        incx = -1;
        delta_x = -delta_x;
    }
    if (delta_y > 0)
        incy = 1;
    else if (delta_y == 0)
        incy = 0; // 水平线
    else
    {
        incy = -1;
        delta_y = -delta_x;
    }
    if (delta_x > delta_y)
        distance = delta_x; // 选取基本增量坐标轴
    else
        distance = delta_y;
    for (t = 0; t < distance + 1; t++)
    {
        OLED_DrawPoint_SPI_HARD(uRow, uCol, mode); // 画点
        xerr += delta_x;
        yerr += delta_y;
        if (xerr > distance)
        {
            xerr -= distance;
            uRow += incx;
        }
        if (yerr > distance)
        {
            yerr -= distance;
            uCol += incy;
        }
    }
}
// x,y:圆心坐标
// r:圆的半径
void OLED_DrawCircle_SPI_HARD(u8 x, u8 y, u8 r)
{
    int a, b, num;
    a = 0;
    b = r;
    while (2 * b * b >= r * r)
    {
        OLED_DrawPoint_SPI_HARD(x + a, y - b, 1);
        OLED_DrawPoint_SPI_HARD(x - a, y - b, 1);
        OLED_DrawPoint_SPI_HARD(x - a, y + b, 1);
        OLED_DrawPoint_SPI_HARD(x + a, y + b, 1);

        OLED_DrawPoint_SPI_HARD(x + b, y + a, 1);
        OLED_DrawPoint_SPI_HARD(x + b, y - a, 1);
        OLED_DrawPoint_SPI_HARD(x - b, y - a, 1);
        OLED_DrawPoint_SPI_HARD(x - b, y + a, 1);

        a++;
        num = (a * a + b * b) - r * r; // 计算画的点离圆心的距离
        if (num > 0)
        {
            b--;
            a--;
        }
    }
}

// 在指定位置显示一个字符,包括部分字符
// x:0~127
// y:0~63
// size1:选择字体 6x8/6x12/8x16/12x24
// mode:0,反色显示;1,正常显示
void OLED_ShowChar_SPI_HARD(u8 x, u8 y, u8 chr, u8 size1, u8 mode)
{
    u8 i, m, temp, size2, chr1;
    u8 x0 = x, y0 = y;
    if (size1 == 8)
        size2 = 6;
    else
        size2 = (size1 / 8 + ((size1 % 8) ? 1 : 0)) * (size1 / 2); // 得到字体一个字符对应点阵集所占的字节数
    chr1 = chr - ' ';                                              // 计算偏移后的值
    for (i = 0; i < size2; i++)
    {
        if (size1 == 8)
        {
            temp = asc2_0806[chr1][i];
        } // 调用0806字体
        else if (size1 == 12)
        {
            temp = asc2_1206[chr1][i];
        } // 调用1206字体
        else if (size1 == 16)
        {
            temp = asc2_1608[chr1][i];
        } // 调用1608字体
        else if (size1 == 24)
        {
            temp = asc2_2412[chr1][i];
        } // 调用2412字体
        else
            return;
        for (m = 0; m < 8; m++)
        {
            if (temp & 0x01)
                OLED_DrawPoint_SPI_HARD(x, y, mode);
            else
                OLED_DrawPoint_SPI_HARD(x, y, !mode);
            temp >>= 1;
            y++;
        }
        x++;
        if ((size1 != 8) && ((x - x0) == size1 / 2))
        {
            x = x0;
            y0 = y0 + 8;
        }
        y = y0;
    }
}

// 显示字符串
// x,y:起点坐标
// size1:字体大小
//*chr:字符串起始地址
// mode:0,反色显示;1,正常显示
void OLED_ShowString_SPI_HARD(u8 x, u8 y, u8 *chr, u8 size1, u8 mode)
{
    while ((*chr >= ' ') && (*chr <= '~')) // 判断是不是非法字符!
    {
        OLED_ShowChar_SPI_HARD(x, y, *chr, size1, mode);
        if (size1 == 8)
            x += 6;
        else
            x += size1 / 2;
        chr++;
    }
}

// m^n
u32 OLED_Pow_SPI_HARD(u8 m, u8 n)
{
    u32 result = 1;
    while (n--)
    {
        result *= m;
    }
    return result;
}

// 显示数字
// x,y :起点坐标
// num :要显示的数字
// len :数字的位数
// size:字体大小
// mode:0,反色显示;1,正常显示
void OLED_ShowNum_SPI_HARD(u8 x, u8 y, u32 num, u8 len, u8 size1, u8 mode)
{
    u8 t, temp, m = 0;
    if (size1 == 8)
        m = 2;
    for (t = 0; t < len; t++)
    {
        temp = (num / OLED_Pow_SPI_HARD(10, len - t - 1)) % 10;
        if (temp == 0)
        {
            OLED_ShowChar_SPI_HARD(x + (size1 / 2 + m) * t, y, '0', size1, mode);
        }
        else
        {
            OLED_ShowChar_SPI_HARD(x + (size1 / 2 + m) * t, y, temp + '0', size1, mode);
        }
    }
}

// 显示汉字
// x,y:起点坐标
// num:汉字对应的序号
// mode:0,反色显示;1,正常显示
void OLED_ShowChinese_SPI_HARD(u8 x, u8 y, u8 num, u8 size1, u8 mode)
{
    u8 m, temp;
    u8 x0 = x, y0 = y;
    u16 i, size3 = (size1 / 8 + ((size1 % 8) ? 1 : 0)) * size1; // 得到字体一个字符对应点阵集所占的字节数
    for (i = 0; i < size3; i++)
    {
        if (size1 == 16)
        {
            temp = Hzk1[num][i];
        } // 调用16*16字体
        else if (size1 == 24)
        {
            temp = Hzk2[num][i];
        } // 调用24*24字体
        else if (size1 == 32)
        {
            temp = Hzk3[num][i];
        } // 调用32*32字体
        else if (size1 == 64)
        {
            temp = Hzk4[num][i];
        } // 调用64*64字体
        else
            return;
        for (m = 0; m < 8; m++)
        {
            if (temp & 0x01)
                OLED_DrawPoint_SPI_HARD(x, y, mode);
            else
                OLED_DrawPoint_SPI_HARD(x, y, !mode);
            temp >>= 1;
            y++;
        }
        x++;
        if ((x - x0) == size1)
        {
            x = x0;
            y0 = y0 + 8;
        }
        y = y0;
    }
}

// num 显示汉字的个数
// space 每一遍显示的间隔
// mode:0,反色显示;1,正常显示
void OLED_ScrollDisplay_SPI_HARD(u8 num, u8 space, u8 mode)
{
    u8 i, n, t = 0, m = 0, r;
    while (1)
    {
        if (m == 0)
        {
            OLED_ShowChinese_SPI_HARD(128, 24, t, 16, mode); // 写入一个汉字保存在OLED_GRAM_SPI_HARD[][]数组中
            t++;
        }
        if (t == num)
        {
            for (r = 0; r < 16 * space; r++) // 显示间隔
            {
                for (i = 1; i < 144; i++)
                {
                    for (n = 0; n < 8; n++)
                    {
                        OLED_GRAM_SPI_HARD[i - 1][n] = OLED_GRAM_SPI_HARD[i][n];
                    }
                }
                OLED_Refresh_SPI_HARD();
            }
            t = 0;
        }
        m++;
        if (m == 16)
        {
            m = 0;
        }
        for (i = 1; i < 144; i++) // 实现左移
        {
            for (n = 0; n < 8; n++)
            {
                OLED_GRAM_SPI_HARD[i - 1][n] = OLED_GRAM_SPI_HARD[i][n];
            }
        }
        OLED_Refresh_SPI_HARD();
    }
}

// x,y：起点坐标
// sizex,sizey,图片长宽
// BMP[]：要写入的图片数组
// mode:0,反色显示;1,正常显示
void OLED_ShowPicture_SPI_HARD(u8 x, u8 y, u8 sizex, u8 sizey, u8 BMP[], u8 mode)
{
    u16 j = 0;
    u8 i, n, temp, m;
    u8 x0 = x, y0 = y;
    sizey = sizey / 8 + ((sizey % 8) ? 1 : 0);
    for (n = 0; n < sizey; n++)
    {
        for (i = 0; i < sizex; i++)
        {
            temp = BMP[j];
            j++;
            for (m = 0; m < 8; m++)
            {
                if (temp & 0x01)
                    OLED_DrawPoint_SPI_HARD(x, y, mode);
                else
                    OLED_DrawPoint_SPI_HARD(x, y, !mode);
                temp >>= 1;
                y++;
            }
            x++;
            if ((x - x0) == sizex)
            {
                x = x0;
                y0 = y0 + 8;
            }
            y = y0;
        }
    }
}
// OLED的初始化
void OLED_Init_SPI_HARD(void)
{
    spi_hard_oled_config();

    OLED_WR_Byte_SPI_HARD(0xAE, OLED_CMD); //--turn off oled panel
    OLED_WR_Byte_SPI_HARD(0x00, OLED_CMD); //---set low column address
    OLED_WR_Byte_SPI_HARD(0x10, OLED_CMD); //---set high column address
    OLED_WR_Byte_SPI_HARD(0x40, OLED_CMD); //--set start line address  Set Mapping RAM Display Start Line (0x00~0x3F)
    OLED_WR_Byte_SPI_HARD(0x81, OLED_CMD); //--set contrast control register
    OLED_WR_Byte_SPI_HARD(0xCF, OLED_CMD); // Set SEG Output Current Brightness
    OLED_WR_Byte_SPI_HARD(0xA1, OLED_CMD); //--Set SEG/Column Mapping     0xa0左右反置 0xa1正常
    OLED_WR_Byte_SPI_HARD(0xC8, OLED_CMD); // Set COM/Row Scan Direction   0xc0上下反置 0xc8正常
    OLED_WR_Byte_SPI_HARD(0xA6, OLED_CMD); //--set normal display
    OLED_WR_Byte_SPI_HARD(0xA8, OLED_CMD); //--set multiplex ratio(1 to 64)
    OLED_WR_Byte_SPI_HARD(0x3f, OLED_CMD); //--1/64 duty
    OLED_WR_Byte_SPI_HARD(0xD3, OLED_CMD); //-set display offset	Shift Mapping RAM Counter (0x00~0x3F)
    OLED_WR_Byte_SPI_HARD(0x00, OLED_CMD); //-not offset
    OLED_WR_Byte_SPI_HARD(0xd5, OLED_CMD); //--set display clock divide ratio/oscillator frequency
    OLED_WR_Byte_SPI_HARD(0x80, OLED_CMD); //--set divide ratio, Set Clock as 100 Frames/Sec
    OLED_WR_Byte_SPI_HARD(0xD9, OLED_CMD); //--set pre-charge period
    OLED_WR_Byte_SPI_HARD(0xF1, OLED_CMD); // Set Pre-Charge as 15 Clocks & Discharge as 1 Clock
    OLED_WR_Byte_SPI_HARD(0xDA, OLED_CMD); //--set com pins hardware configuration
    OLED_WR_Byte_SPI_HARD(0x12, OLED_CMD);
    OLED_WR_Byte_SPI_HARD(0xDB, OLED_CMD); //--set vcomh
    OLED_WR_Byte_SPI_HARD(0x40, OLED_CMD); // Set VCOM Deselect Level
    OLED_WR_Byte_SPI_HARD(0x20, OLED_CMD); //-Set Page Addressing Mode (0x00/0x01/0x02)
    OLED_WR_Byte_SPI_HARD(0x02, OLED_CMD); //
    OLED_WR_Byte_SPI_HARD(0x8D, OLED_CMD); //--set Charge Pump enable/disable
    OLED_WR_Byte_SPI_HARD(0x14, OLED_CMD); //--set(0x10) disable
    OLED_WR_Byte_SPI_HARD(0xA4, OLED_CMD); // Disable Entire Display On (0xa4/0xa5)
    OLED_WR_Byte_SPI_HARD(0xA6, OLED_CMD); // Disable Inverse Display On (0xa6/a7)
    OLED_Clear_SPI_HARD();
    OLED_WR_Byte_SPI_HARD(0xAF, OLED_CMD);
}
